#include "iniReader.hpp"
#include <fstream>
#include <stdlib.h>

bool IniReader::IsSpace(char c)
{
    if (' ' == c || '\t' == c)
        return true;
    return false;
}

bool IniReader::IsCommentChar(char c)
{
    switch (c) {
    case '#':
        return true;
    default:
        return false;
    }
}

void IniReader::Trim(std::string & str)
{
    if (str.empty())
    {
        return;
    }
    int i, start_pos, end_pos;
    for (i = 0; i < str.size(); ++i) {
        if (!IsSpace(str[i])) {
            break;
        }
    }
    if (i == str.size())
    {
        str = "";
        return;
    }
    start_pos = i;
    for (i = str.size() - 1; i >= 0; --i) {
        if (!IsSpace(str[i])) {
            break;
        }
    }
    end_pos = i;
    str = str.substr(start_pos, end_pos - start_pos + 1);
}

bool IniReader::AnalyseLine(const std::string & line, std::string& section, std::string & key, std::string & value)
{
    if (line.empty())
        return false;
    int start_pos = 0, end_pos = line.size() - 1, pos, s_startpos, s_endpos;
    if ((pos = line.find("#")) != -1)
    {
        if (0 == pos)
        {
            return false;
        }
        end_pos = pos - 1;
    }
    if (((s_startpos = line.find("[")) != -1) && ((s_endpos = line.find("]"))) != -1)
    {
        section = line.substr(s_startpos + 1, s_endpos - 1);
        return true;
    }
    std::string new_line = line.substr(start_pos, start_pos + 1 - end_pos);
    if ((pos = new_line.find('=')) == -1)
        return false;
    key = new_line.substr(0, pos);
    value = new_line.substr(pos + 1, end_pos + 1 - (pos + 1));
    Trim(key);
    if (key.empty()) {
        return false;
    }
    Trim(value);
    if ((pos = value.find("\r")) > 0)
    {
        value.replace(pos, 1, "");
    }
    if ((pos = value.find("\n")) > 0)
    {
        value.replace(pos, 1, "");
    }
    return true;
}

bool IniReader::ReadConfig(const std::string & filename)
{
    settings_.clear();
    std::ifstream infile(filename.c_str());
    if (!infile.is_open()) {
        return false;
    }
    std::string line, key, value, section;
    std::map<std::string, std::string> k_v;
    std::map<std::string, std::map<std::string, std::string> >::iterator it;
    while (getline(infile, line))
    {
        if (AnalyseLine(line, section, key, value))
        {
            it = settings_.find(section);
            if (it != settings_.end())
            {
                k_v[key] = value;
                it->second = k_v;
            }
            else
            {
                k_v.clear();
                settings_.insert(std::make_pair(section, k_v));
            }
        }
        key.clear();
        value.clear();
    }
    infile.close();
    return true;
}

std::string IniReader::ReadString(const char* section, const char* item, const char* default_value)
{
    std::string tmp_s(section);
    std::string tmp_i(item);
    std::string def(default_value);
    std::map<std::string, std::string> k_v;
    std::map<std::string, std::string>::iterator it_item;
    std::map<std::string, std::map<std::string, std::string> >::iterator it;
    it = settings_.find(tmp_s);
    if (it == settings_.end())
    {
        return def;
    }
    k_v = it->second;
    it_item = k_v.find(tmp_i);
    if (it_item == k_v.end())
    {
        return def;
    }
    return it_item->second;
}

std::string IniReader::ReadString(const char* section, const char* item)
{
    std::string tmp_s(section);
    std::string tmp_i(item);
    std::string def("");
    std::map<std::string, std::string> k_v;
    std::map<std::string, std::string>::iterator it_item;
    std::map<std::string, std::map<std::string, std::string> >::iterator it;
    it = settings_.find(tmp_s);
    if (it == settings_.end())
    {
        return def;
    }
    k_v = it->second;
    it_item = k_v.find(tmp_i);
    if (it_item == k_v.end())
    {
        return def;
    }
    return it_item->second;
}

int IniReader::ReadInt(const char* section, const char* item, const int& default_value)
{
    std::string tmp_s(section);
    std::string tmp_i(item);
    std::map<std::string, std::string> k_v;
    std::map<std::string, std::string>::iterator it_item;
    std::map<std::string, std::map<std::string, std::string> >::iterator it;
    it = settings_.find(tmp_s);
    if (it == settings_.end())
    {
        return default_value;
    }
    k_v = it->second;
    it_item = k_v.find(tmp_i);
    if (it_item == k_v.end())
    {
        return default_value;
    }
    return atoi(it_item->second.c_str());
}

float IniReader::ReadFloat(const char* section, const char* item, const float& default_value)
{
    std::string tmp_s(section);
    std::string tmp_i(item);
    std::map<std::string, std::string> k_v;
    std::map<std::string, std::string>::iterator it_item;
    std::map<std::string, std::map<std::string, std::string> >::iterator it;
    it = settings_.find(tmp_s);
    if (it == settings_.end())
    {
        return default_value;
    }
    k_v = it->second;
    it_item = k_v.find(tmp_i);
    if (it_item == k_v.end())
    {
        return default_value;
    }
    return atof(it_item->second.c_str());
}

